torch
torchvision
torchaudio
transformers
sentencepiece
datasets
bitsandbytes
accelerate
